import os
import glob


def find_cuda_path():
    # 检查常见的安装路径
    possible_paths = [
        r"C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v*",
        r"D:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v*",
        r"D:\NVIDIA\CUDA\v*",
        r"D:\CUDA\v*",
    ]

    for pattern in possible_paths:
        paths = glob.glob(pattern)
        if paths:
            print("找到的 CUDA 路径:")
            for path in paths:
                print(path)
            return

    print("未找到 CUDA 安装路径")


if __name__ == "__main__":
    find_cuda_path()
